super mario python code